iT邦幫忙

2023 iThome 鐵人賽

DAY 11
0
自我挑戰組

leetcode題目分享系列 第 11

[Day 11] 1282. Group the People Given the Group Size They Belong To

  • 分享至 

  • xImage
  •  

這題主要使用Hashmap,將相同size的人放在同一格hash裡,再從裡面算組數丟進答案中

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        unordered_map<int, vector<int>> umap;
        for(int i = 0; i < groupSizes.size(); i++){
            umap[groupSizes[i]].push_back(i);
        }
        vector<vector<int>> ans;
        for(auto [x, v] : umap){
            int q = umap[x].size() / x;
            for(int i = 0; i < q; i++){
                vector<int> tmp;
                for(int j = 0; j < x; j++){
                    tmp.push_back(umap[x][j + i * x]);
                }
                ans.push_back(tmp);
            } 
            
        }
        return ans;
    }
};

上一篇
[Day 10] 1359. Count All Valid Pickup and Delivery Options
下一篇
[Day 12] 1647. Minimum Deletions to Make Character Frequencies Unique
系列文
leetcode題目分享30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言